# -*- coding: utf-8 -*-
"""
@Auth ： laichenying
@File ：parse_excel.py
@IDE ：PyCharm

"""
from openpyxl import load_workbook


class ParseExcel(object):
    def __init__(self, excelPath, sheetName):
        print(excelPath, sheetName)
        self.wb = load_workbook(excelPath)
        self.sheet = self.wb[sheetName]
        self.maxRowNum = self.sheet.max_row

    def getDatasFromSheet(self):
        dataList = []
        for line in self.sheet.rows:
            tmpList = []
            # 根据Excel文件每一竖列，有需要的就需要获取，没需要的就不需要获取,故line[i]为获取的列号
            tmpList.append(line[1].value)
            tmpList.append(line[5].value)
            tmpList.append(line[6].value)
            dataList.append(tmpList)
        # 头两行为标题列，不获取数据，故dataList[2:]
        return dataList[2:]

    def getDatasFromSheet_smoke(self):
        dataList = []
        for line in self.sheet.rows:
            tmpList = []
            # 根据Excel文件每一竖列，有需要的就需要获取，没需要的就不需要获取,故line[i]为获取的列号
            tmpList.append(line[1].value)
            tmpList.append(line[4].value)
            tmpList.append(line[5].value)
            tmpList.append(line[6].value)
            dataList.append(tmpList)

        # 头两行为标题列，不获取数据，故dataList[2:]
        return dataList[2:]

    def getDatasFromSheet_01(self):
        dataList = []
        for line in self.sheet.rows:
            tmpList = []
            # 根据Excel文件每一竖列，有需要的就需要获取，没需要的就不需要获取,故line[i]为获取的列号
            tmpList.append(line[1].value)
            tmpList.append(line[5].value)
            tmpList.append(line[6].value)
            tmpList.append(line[7].value)
            dataList.append(tmpList)
        # 头两行为标题列，不获取数据，故dataList[2:]
        return dataList[2:]

    def getDatasFromSheet_smoke_01(self):
        dataList = []
        for line in self.sheet.rows:
            tmpList = []
            for line in self.sheet.rows:
                tmpList = []
            # 根据Excel文件每一竖列，有需要的就需要获取，没需要的就不需要获取,故line[i]为获取的列号
                tmpList.append(line[1].value)
                tmpList.append(line[4].value)
                tmpList.append(line[5].value)
                tmpList.append(line[6].value)
                tmpList.append(line[7].value)
                dataList.append(tmpList)
        # 头两行为标题列，不获取数据，故dataList[2:]
        return dataList[2:]

    def getDatasFromSheet_auditTask(self):
        dataList = []
        for line in self.sheet.rows:
            tmpList = []
            for line in self.sheet.rows:
                tmpList = []
            # 根据Excel文件每一竖列，有需要的就需要获取，没需要的就不需要获取,故line[i]为获取的列号
                tmpList.append(line[1].value)
                tmpList.append(line[4].value)
                tmpList.append(line[5].value)
                tmpList.append(line[6].value)
                dataList.append(tmpList)
        # 头两行为标题列，不获取数据，故dataList[2:]
        return dataList[2:]


    def getDatasFromSheet_firewareUpdate(self):
        dataList = []
        for line in self.sheet.rows:
            tmpList = []
            for line in self.sheet.rows:
                tmpList = []
            # 根据Excel文件每一竖列，有需要的就需要获取，没需要的就不需要获取,故line[i]为获取的列号
                tmpList.append(line[1].value)
                tmpList.append(line[4].value)
                tmpList.append(line[5].value)
                tmpList.append(line[6].value)
                dataList.append(tmpList)
        # 头两行为标题列，不获取数据，故dataList[2:]
        return dataList[2:]